/**
 * Software License, Version 1.0
 * 
 * Copyright 2003 The Trustees of Indiana University.  All rights reserved.
 * 
 *
 *Redistribution and use in source and binary forms, with or without 
 *modification, are permitted provided that the following conditions are met:
 *
 *1) All redistributions of source code must retain the above copyright notice,
 * the list of authors in the original source code, this list of conditions and
 * the disclaimer listed in this license;
 *2) All redistributions in binary form must reproduce the above copyright 
 * notice, this list of conditions and the disclaimer listed in this license in
 * the documentation and/or other materials provided with the distribution;
 *3) Any documentation included with all redistributions must include the 
 * following acknowledgement:
 *
 *"This product includes software developed by the Community Grids Lab. For 
 * further information contact the Community Grids Lab at 
 * http://communitygrids.iu.edu/."
 *
 * Alternatively, this acknowledgement may appear in the software itself, and 
 * wherever such third-party acknowledgments normally appear.
 * 
 *4) The name Indiana University or Community Grids Lab or NaradaBrokering, 
 * shall not be used to endorse or promote products derived from this software 
 * without prior written permission from Indiana University.  For written 
 * permission, please contact the Advanced Research and Technology Institute 
 * ("ARTI") at 351 West 10th Street, Indianapolis, Indiana 46202.
 *5) Products derived from this software may not be called NaradaBrokering, 
 * nor may Indiana University or Community Grids Lab or NaradaBrokering appear
 * in their name, without prior written permission of ARTI.
 * 
 *
 * Indiana University provides no reassurances that the source code provided 
 * does not infringe the patent or any other intellectual property rights of 
 * any other entity.  Indiana University disclaims any liability to any 
 * recipient for claims brought by any other entity based on infringement of 
 * intellectual property rights or otherwise.  
 *
 *LICENSEE UNDERSTANDS THAT SOFTWARE IS PROVIDED "AS IS" FOR WHICH NO 
 *WARRANTIES AS TO CAPABILITIES OR ACCURACY ARE MADE. INDIANA UNIVERSITY GIVES
 *NO WARRANTIES AND MAKES NO REPRESENTATION THAT SOFTWARE IS FREE OF 
 *INFRINGEMENT OF THIRD PARTY PATENT, COPYRIGHT, OR OTHER PROPRIETARY RIGHTS. 
 *INDIANA UNIVERSITY MAKES NO WARRANTIES THAT SOFTWARE IS FREE FROM "BUGS", 
 *"VIRUSES", "TROJAN HORSES", "TRAP DOORS", "WORMS", OR OTHER HARMFUL CODE.  
 *LICENSEE ASSUMES THE ENTIRE RISK AS TO THE PERFORMANCE OF SOFTWARE AND/OR 
 *ASSOCIATED MATERIALS, AND TO THE PERFORMANCE AND VALIDITY OF INFORMATION 
 *GENERATED USING SOFTWARE.
 */
package cgl.narada.service.security.events;

import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;
import java.io.DataInputStream;
import java.io.DataOutputStream;
import java.io.IOException;

public class TemplateKeyResponse {
  private int entityId;
  private int templateId;
  private boolean success;
  private String reason;
  private byte[] encryptedTemplateKey;

  private String moduleName;

  public TemplateKeyResponse(int entityId, int templateId,
			     byte[] encryptedTemplateKey) {
    this.entityId = entityId;
    this.templateId = templateId;
    success = true;
    this.encryptedTemplateKey = encryptedTemplateKey;
  }

  /** Called in response to a denial in the request to= retrieve a template
      key */
  public TemplateKeyResponse(int entityId, int templateId, String reason) {
    this.entityId = entityId;
    this.templateId = templateId;
    success = false;
    if (reason == null) {
      this.reason = "Failure";
    } else {
      this.reason = reason;
    }
  }

  
  public int getEntityId() {
    return entityId;
  }


  public int getTemplateId() {
    return templateId;
  }

  
  public byte[] getEncryptedTemplateKey() {
    return encryptedTemplateKey;
  }

  
  public boolean isSuccess() {
    return success;
  }

  public String getFailureReason() {
    return reason;
  }

  public byte[] getBytes() {
    ByteArrayOutputStream baOutputStream = new ByteArrayOutputStream();
    DataOutputStream dout = new DataOutputStream(baOutputStream);
    byte[] marshalledBytes= null;
    try {
      dout.writeByte(SecureExchange.TEMPLATE_KEY_RESPONSE);
      dout.writeInt(entityId);
      dout.writeInt(templateId);
      
      dout.writeBoolean(success);
      
      if (success) {
	dout.writeInt(encryptedTemplateKey.length);
	dout.write(encryptedTemplateKey);
      } else {
	dout.writeInt(reason.length());
	dout.write(reason.getBytes());	
      }
      dout.flush();
      marshalledBytes = baOutputStream.toByteArray();
    } catch (IOException ioe) {
      System.out.println(moduleName + "Error while marshalling");
    } /** end try-catch block*/
    
    return marshalledBytes;
  }
  
  public TemplateKeyResponse(byte[] marshalledBytes) {
    ByteArrayInputStream baInputStream = 
      new ByteArrayInputStream(marshalledBytes);
    DataInputStream din = new DataInputStream(baInputStream);

    try {
      din.readByte(); /** SecureExchange.TEMPLATE_KEY_RESPONSE*/
      entityId = din.readInt();
      templateId = din.readInt();

      success = din.readBoolean();
      
      if (success) {
	encryptedTemplateKey = new byte[din.readInt()];
	din.readFully(encryptedTemplateKey);
      } else {
	byte[] reasonBytes = new byte[din.readInt()];
	din.readFully(reasonBytes);
	reason = new String(reasonBytes);
      }
      
      
    } catch (IOException ioe) {
      System.out.println(moduleName + "Error while un-marshalling");
    }/** end try-catch block*/

  }

}
